# -*- coding: utf-8 -*-

import os


cdir = os.path.dirname(__file__)
model_dir = os.path.join(cdir, 'models')
cluster_one_model = os.path.join(model_dir, 'cluster_one.pkl')
cluster_two_model = os.path.join(model_dir, 'cluster_two.pkl')
cluster_three_model = os.path.join(model_dir, 'cluster_three.pkl')

svm_model = os.path.join(model_dir, 'svm_model.pkl')

param_dir = os.path.join(cdir, 'models', 'params')
cluster_gaussian_params = os.path.join(param_dir, 'cluster_gaussian_param.pkl')

if not os.path.exists(model_dir):
    os.makedirs(model_dir)
if not os.path.exists(param_dir):
    os.makedirs(param_dir)

